import matplotlib.pyplot as plt
import numpy as np
from scipy import signal

ws=np.linspace(0,10,512)

num=np.array([1,0,0])
den=np.array([1,2,1])

tout,yout=signal.impulse((num,den))
w,h=signal.freqs(num,den,worN=ws)

plt.rcParams['font.sans-serif']=['SimHei']#加上这一句就能在图表中显示中文
plt.rcParams['axes.unicode_minus']=False#用来正常显示负号

plt.subplot(211)
plt.plot(tout,yout)
plt.xlabel('t(s)')
plt.ylabel('h(s)')
plt.title('单位冲激响应')
plt.subplot(212)
plt.plot(w,np.abs(h))
plt.title('幅频')
plt.xlabel(r'$\omega(rad/s)$')
plt.show()
